package info.wangyuan.agent.entity.po;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import info.wangyuan.agent.common.Enum.SensitiveType;
import info.wangyuan.agent.common.annotation.Sensitive;
import lombok.*;

import java.io.Serial;
import java.io.Serializable;

/**
 * 模型配置实体类
 *
 * @author Albert
 * @since 2025-07-10 02:24:44
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
@EqualsAndHashCode(callSuper = true)
@TableName(value = "tb_models", autoResultMap = true)
public class ModelConfig extends BaseEntity implements Serializable {
    @Serial
    private static final long serialVersionUID = 1L;

    @TableId(type = IdType.AUTO)
    private Integer id;

    /**
     * 模型名称
     */
    private String name;

    /**
     * 请求基地址
     */
    private String baseUrl;

    /**
     * 密钥（脱敏处理）
     */
    @Sensitive(type = SensitiveType.API_KEY)
    private String apiKey;

    /**
     * 模型类型
     */
    private String type;

    /**
     * 模型活跃状态（默认未激活）
     */
    @Builder.Default
    private Boolean status = Boolean.FALSE;

    /**
     * 服务商ID
     */
    private Integer providerId;

    /**
     * 模型温度（默认0.50）
     */
    @Builder.Default
    private Double temperature = 0.50;

    /**
     * 是否返回思考过程（默认不返回）
     */
    @Builder.Default
    private Boolean returnThinking = Boolean.FALSE;

    /**
     * 备注
     */
    private String remark;
}
